package com.heima.app.gateway.filter;

import com.heima.utils.AppJwtUtil;
import io.jsonwebtoken.Claims;
import org.apache.commons.lang3.StringUtils;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

import java.util.function.Consumer;
@Component
public class AuthorizeFilter implements Ordered, GlobalFilter {
	@Override
	public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
		ServerHttpRequest request = exchange.getRequest();
		ServerHttpResponse response = exchange.getResponse();
		//如果包含登录路径则直接放行
		if (request.getURI().getPath().contains("/login")) {
			return chain.filter(exchange);
		}
		String token = request.getHeaders().getFirst("token");
		//token不存在,直接返回
		if (StringUtils.isBlank(token)) {
			response.setStatusCode(HttpStatus.UNAUTHORIZED);
			return response.setComplete();
		}
		
		try {
			Claims claimsBody = AppJwtUtil.getClaimsBody(token);
			int result = AppJwtUtil.verifyToken(claimsBody);
			if (result == 1 || result == 2) {
				response.setStatusCode(HttpStatus.UNAUTHORIZED);
				return response.setComplete();
			}
		} catch (Exception e) {
			e.printStackTrace();
			response.setStatusCode(HttpStatus.UNAUTHORIZED);
			return response.setComplete();
		}
		
		try {
			Claims claimsBody = AppJwtUtil.getClaimsBody(token);
			//是否是过期
			int result = AppJwtUtil.verifyToken(claimsBody);
			if(result == 1 || result  == 2){
				response.setStatusCode(HttpStatus.UNAUTHORIZED);
				return response.setComplete();
			}
			//如果token没有过期则获取用户id
			
			Long id = new Long((Integer)claimsBody.get("id"));
			//获取到id后刷新到请求头上
			ServerHttpRequest newRequest = request.mutate().headers(new Consumer<HttpHeaders>() {
				@Override
				public void accept(HttpHeaders httpHeaders) {
					httpHeaders.add("userId", id.toString());
				}
			}).build();
			
			//将构建好的新请求刷新到交换器上.
			exchange.mutate().request(newRequest).build();
		} catch (Exception e) {
			e.printStackTrace();
		}
		
		return chain.filter(exchange);
	}
	
	@Override
	public int getOrder() {
		return 0;
	}
}
